Feature engineering key error in python